import argparse


def parse_args():
    parser = argparse.ArgumentParser(description="Parameters of preprocessing and statistical program")
    parser.add_argument('--src-dir', type=str, default='G:\\', help='Storage path of Deepfake images')
    parser.add_argument('--dst-dir', type=str, default='E:\\Datasets\\test',
                        help='Destination path of Deepfake images')
    parser.add_argument("--fake-type", type=str, default="Deepfakes",
                        choices=['Deepfakes', 'Face2Face', 'FaceShifter', 'FaceSwap', 'NeuralTextures'],
                        help="Manipulation type")
    args = parser.parse_args()
    return args
